package com.ruoyi;

import org.eclipse.paho.client.mqttv3.*;
import org.eclipse.paho.client.mqttv3.persist.MemoryPersistence;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @Author Alan Zhang
 * @Date 2025/3/12 10:08
 */
@Configuration
public class MqttClientConfig {
    private static final Logger logger = LoggerFactory.getLogger(MqttClientConfig.class);
    @Value("${spring.mqtt.username}")
    private String username;

    @Value("${spring.mqtt.password}")
    private String password;

    @Value("${spring.mqtt.url}")
    private String hostUrl;

    @Value("${spring.mqtt.client.id}")
    private String clientId;

    private MqttClientCallBack mqttClientCallBack;

    /**
     * 客户端对象
     */
    private MqttAsyncClient client;

    /**
     * 在bean初始化后连接到服务器
     */
    @PostConstruct
    public void init() {
        connect();
    }

    /**
     * 客户端连接服务端
     */
    public void connect() {
        //连接设置
        MqttConnectOptions options = new MqttConnectOptions();
        //是否清空session，设置false表示服务器会保留客户端的连接记录（订阅主题，qos）,客户端重连之后能获取到服务器在客户端断开连接期间推送的消息
        //设置为true表示每次连接服务器都是以新的身份
        options.setCleanSession(false);
        //设置连接用户名
        options.setUserName(username);
        //设置连接密码
        options.setPassword(password.toCharArray());
        //设置超时时间，单位为秒
        options.setConnectionTimeout(60);
        //设置心跳时间 单位为秒，表示服务器每隔 1.5*10秒的时间向客户端发送心跳判断客户端是否在线
        options.setKeepAliveInterval(20);
        // 开启自动重连
        options.setAutomaticReconnect(true);
        // 设置最大重连时间间隔 (可选)，单位是毫秒，设置为 5000 表示最多等待 5 秒再尝试重连
        options.setMaxReconnectDelay(5000);
        //设置遗嘱消息的话题，若客户端和服务器之间的连接意外断开，服务器将发布客户端的遗嘱信息
        //QoS服务质量，QoS0至多一次，QoS1至少一次，QoS2只有一次
        options.setWill("willTopic", (clientId + "与服务器断开连接").getBytes(), 0, false);
        try {
            //创建MQTT客户端对象
            client = new MqttAsyncClient(hostUrl, clientId, new MemoryPersistence());
            //设置回调
            mqttClientCallBack = new MqttClientCallBack();

            client.setCallback(mqttClientCallBack);
            // 使用异步连接
            client.connect(options, null, new IMqttActionListener() {
                @Override
                public void onSuccess(IMqttToken asyncActionToken) {
                    logger.info("MQTT连接成功");
                }
                @Override
                public void onFailure(IMqttToken asyncActionToken, Throwable exception) {
                    logger.error("MQTT连接失败：" + exception.getMessage());
                }
            });
        } catch (MqttException e) {
            logger.error("mqtt连接失败。。" + e.getMessage());
        }
    }

    public void publish(int qos, boolean retained, String topic,String message) {

        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(qos);
        mqttMessage.setRetained(retained);
        mqttMessage.setPayload(message.getBytes());
        try {
            // 使用异步客户端发布消息，并处理结果
            client.publish(topic, mqttMessage, null, new IMqttActionListener() {
                @Override
                public void onSuccess(IMqttToken asyncActionToken) {
                    System.out.println("发送成功");
                }
                @Override
                public void onFailure(IMqttToken asyncActionToken, Throwable exception) {
                    logger.error("发送失败：" + exception.getMessage());
                }
            });
        } catch (MqttException e) {
            logger.error("发送失败：" + e.getMessage());
        }
    }

//    public void publish(String topic, String message, int qos, boolean retained) {
//        MqttMessage mqttMessage = new MqttMessage();
//        mqttMessage.setQos(qos);
//        mqttMessage.setRetained(retained);
//        mqttMessage.setPayload(message.getBytes());
//        MqttTopic mqttTopic = client.getTopic(topic);
//        //提供一种机制跟踪消息的传递进度，以非阻塞方式执行发布时跟踪消息的传递进度
//        MqttDeliveryToken token;
//        try {
//            //将指定消息发布到主题，但不等待消息传递完成，返回的token可用于跟踪消息的传递状态
//
//            token = mqttTopic.publish(mqttMessage);
//            token.waitForCompletion();
//        } catch (MqttException e) {
//            logger.info("Mqtt publish failed.");
//        }
//    }

    /**
     * 断开连接
     */
    public void disConnect() {
        try {
            client.disconnect();
        } catch (MqttException e) {
            e.printStackTrace();
        }
    }
}
